#!/bin/sh

# 设置环境变量
export CUDA_PATH=/usr/local/cuda
export PATH=$CUDA_PATH/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH

# 打印环境变量以确认设置
echo "CUDA_PATH is set to: $CUDA_PATH"
echo "LD_LIBRARY_PATH is set to: $LD_LIBRARY_PATH"
echo "PATH is set to: $PATH"
echo "nvcc --version"
nvcc --version
